# 从本地获取基础镜像
FROM ascendhub.huawei.com/public-ascendhub/pytorch-modelzoo:22.0.0-1.8.1

# 指定工作目录
WORKDIR /home/HwHiAiUser

# 使用root，安装会快些
USER root

# 安装依赖
COPY requirements.txt .
RUN python3.7 -m pip install --upgrade pip && \
    python3.7 -m pip install -r requirements.txt && \
    python3.7 -m pip install Pillow==9.1.0 

# 切换回默认用户
USER HwHiAiUser
